import os
import shutil
 
img_dir = "nuScenes_yolor_d6_3rd_infer_45k"
xml_dir = "nuscenes_all_txt_1232786_infer_3rd_epoch120"
xml_out_dir = "nuScenes_yolor_d6_3rd_infer_45k_txt"
if not os.path.exists(xml_out_dir):
    os.makedirs(xml_out_dir)

if os.path.exists(img_dir):
    for root, dirs, files in os.walk(img_dir):
        #print("root:",root)#文件夹路径
        #print("dirs:",dirs)#文件夹名称
        #print("files:",files)#文件名
        
        for file in files:#遍历每一个文件
            filename = os.path.basename(file)#获取文件名
            print(filename)
            filename = filename[:-4]
            
            xml_file = filename + '.txt'#拼接得到文件路径
            if os.path.exists(os.path.join(xml_dir, xml_file)):
                print(xml_file)
                #shutil.move(xml_file, xml_dir)#移动文件
                oldpath = os.path.join(xml_dir, xml_file)
                shutil.copy(oldpath, xml_out_dir)
            else:
                continue